package com.lagou.edu.controller;

import com.sun.mail.util.MailSSLSocketFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.mail.*;
import javax.mail.internet.InternetAddress;
import javax.mail.internet.MimeMessage;
import java.security.GeneralSecurityException;
import java.util.Properties;


@RestController
@RequestMapping("/email")
public class EmailController {

    @Value("${email.username}")
    private String userName;
    @Value("${email.password}")
    private String password;
    @Value("${email.message}")
    private String messageText;


    /**
     * 发送邮箱验证码
     *
     * @return
     */
    @GetMapping("/{email}/{code}")
    public Boolean sendEmail(@PathVariable(value = "email") String email,
                             @PathVariable(value = "code") String code) throws GeneralSecurityException {
        sendEmailCode(email, code);
        return true;
    }

    public void sendEmailCode(String email, String code) throws GeneralSecurityException {
        // 收件人电子邮箱
        String to = email;

        // 发件人电子邮箱
        String from = "dchuan680@163.com";

        // 指定发送邮件的主机为 smtp.qq.com
        String host = "smtp.163.com";  //QQ 邮件服务器

        // 获取系统属性
        Properties properties = System.getProperties();

        // 设置邮件服务器
        properties.setProperty("mail.smtp.host", host);

        properties.put("mail.smtp.auth", "true");
        MailSSLSocketFactory sf = new MailSSLSocketFactory();
        sf.setTrustAllHosts(true);
        properties.put("mail.smtp.ssl.enable", "true");
        properties.put("mail.smtp.ssl.socketFactory", sf);
        // 获取默认session对象
        Session session = Session.getDefaultInstance(properties, new Authenticator() {
            public PasswordAuthentication getPasswordAuthentication() {
                return new PasswordAuthentication(userName, password); //发件人邮件用户名、密码
            }
        });

        try {
            // 创建默认的 MimeMessage 对象
            MimeMessage message = new MimeMessage(session);

            // Set From: 头部头字段
            message.setFrom(new InternetAddress(from));

            // Set To: 头部头字段
            message.addRecipient(Message.RecipientType.TO, new InternetAddress(to));

            // Set Subject: 头部头字段
            message.setSubject("This is the Subject Line!");

            // 设置消息体
            message.setText(code + messageText);

            // 发送消息
            Transport.send(message);
            System.out.println("Sent message successfully....from runoob.com");
        } catch (Exception mex) {
            mex.printStackTrace();
        }
    }


}
